"""
自定义的分页组件
以后如果想使用这个分页组件，你需要做如下几件事：

导入
    from app01.utils.pagination import Pagination
视图函数中：
    def pretty_list(request):
        # 1，根据自己的情况去帅选自己的数据
        queryset = models.PrettyNum.objects.all()

        # 2，实例化分页对象
        page_object = Pagination(request, queryset)

        context = {
            'search_data': search_data,

            "queryset": page_object.page_queryset,  # 分完页的数据
            'page_string': page_object.html()  # 生成的页码
        }

        # 把数据传到html页面
        return render(request, 'pretty_list.html', context)

在HTML文件中:
    数据自己

        <ul class="pagination">
            {{ page_string }}
        </ul>

"""
from django.utils.safestring import mark_safe


class Pagination(object):

    def __init__(self, request, queryset, page_size=10, page_param='page', plus=5):
        """
        :param request:     请求的对象
        :param queryset:    查询符合条件的数据(根据这个数据给他进行分页处理)
        :param page_size:   每页显示多少数据
        :param page_param:  在URL中传递的获取分页的参数 列如 ?page=22
        :param plus:        显示当前页的 前和后几页（页码-）
        """

        import copy
        query_dict = copy.deepcopy(request.GET)  # 深拷贝
        query_dict._mutable = True  # 改为允许修改
        self.query_dict = query_dict

        self.page_param = page_param
        # query_dict.setlist('page', [11])

        page = request.GET.get(page_param, '1')
        # 判断拿到的页码page是否是个十进制数字
        if page.isdecimal():
            # 如果是转换为整形
            page = int(page)
        else:
            # 否则拿到的就是中文就让page默认为1
            page = 1
        self.page = page  # 获取到当前页码
        self.page_size = page_size  # 每页显示的页码数

        self.start = (page - 1) * page_size  # 起始页码数
        self.end = page * page_size  # 结束页码数

        self.page_queryset = queryset[self.start:self.end]  # 数据库搜索显示

        # 数据总条数
        total_count = queryset.count()
        # 总页码
        total_page_count, div = divmod(total_count, page_size)  # (除数，被除数) 得（商，余数）
        if div:  # 判断余数是否大于1 大于1 页码数加1
            total_page_count += 1
        # 计算出总页码的数
        self.total_page_count = total_page_count

        self.plus = plus


    def html(self):
        # 计算出应该显示的页码【当前页的前后五页

        # 判断数据库中的数据比较少的时候 没有到达11页那就不同
        if self.total_page_count <= 2 * self.plus + 1:
            start_page = 1
            end_page = self.total_page_count
        else:
            # 数据库中的数据超过11页
            # 当前页数大于5时（小的极值）
            if self.page <= self.plus:
                start_page = 1
                end_page = 2 * self.plus + 1
            else:
                # 当后页数超过总页数时
                if (self.page + self.plus) > self.total_page_count:
                    start_page = self.total_page_count - 2 * self.plus
                    end_page = self.total_page_count
                else:
                    start_page = self.page - self.plus
                    end_page = self.page + self.plus

        # 页码
        page_str_list = []

        self.query_dict.setlist(self.page_param, [1])
        # print(self.query_dict.urlencode())

        page_str_list.append('<li><a href="?{}">首页</a></li>'.format(self.query_dict.urlencode()))

        # 上一页
        if self.page > 1:
            self.query_dict.setlist(self.page_param, [self.page - 1])
            prev = '<li><a href="?{}">«</a></li>'.format(self.query_dict.urlencode())
        else:
            self.query_dict.setlist(self.page_param, [1])
            prev = '<li><a href="?page={}">«</a></li>'.format(self.query_dict.urlencode())
        page_str_list.append(prev)
        # le = len(models.PrettyNum.objects.all())
        # for i in range(1, le // 10 + 2):
        # 列表页面（数字）
        for i in range(start_page, end_page + 1):
            self.query_dict.setlist(self.page_param, [i])
            # 给当前页加上标志
            if i == self.page:
                ele = '<li class="active"><a href="?{}">{}</a></li>'.format(self.query_dict.urlencode(), i)
            else:
                ele = '<li><a href="?{}">{}</a></li>'.format(self.query_dict.urlencode(), i)
            page_str_list.append(ele)

        # 下一页
        if self.page < self.total_page_count:
            self.query_dict.setlist(self.page_param, [self.page + 1])
            prev = '<li><a href="?{}">»</a></li>'.format(self.query_dict.urlencode())
        else:
            self.query_dict.setlist(self.page_param, [self.total_page_count])
            prev = '<li><a href="?{}">»</a></li>'.format(self.query_dict.urlencode())
        page_str_list.append(prev)

        # 尾页
        self.query_dict.setlist(self.page_param, [self.total_page_count])
        page_str_list.append('<li><a href="?{}">尾页</a></li>'.format(self.query_dict.urlencode()))

        # 输入页码
        search_string = """
                    <li style="float: right">
                        <form class="form-inline" method="get">
                            <div class="form-group">
                                <input style="width: 80px" type="text" name="page" class="form-control" id="exampleInputEmail3"
                                       placeholder="输入页码">
                            </div>
                            <button type="submit" class="btn btn-default">
                                <span class="glyphicon glyphicon-search" aria-hidden="true"></span>
                            </button>
                        </form>
                    </li>
            """
        page_str_list.append(search_string)
        page_string = mark_safe("".join(page_str_list))
        return page_string
